#!/usr/bin/python3
# -*- coding: utf-8 -*-

import pandas as pd
import numpy  as np
import talib


def signal(*args):
    df = args[0]
    n  = args[1]
    factor_name = args[2]

    # MAAMT 指标
    """
    N=40
    MAAMT=MA(AMOUNT,N)
    MAAMT 是成交额的移动平均线。当成交额上穿/下穿移动平均线时产
    生买入/卖出信号。
    """
    MAAMT = df['volume'].rolling(n, min_periods=1).mean()
    df[factor_name] = (df['volume'] - MAAMT) / MAAMT  # 避免量纲问题

    return df